package com.lfs.aspectj;

import com.lfs.annotation.DataSource;
import com.lfs.util.DynamicDataSourceContextHolder;
import com.lfs.util.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.util.Objects;

@Aspect
@Order(1)
@Component
public class DataSourceAspect {

    @Pointcut("@annotation(com.lfs.annotation.DataSource)")
    public void dsPoint(){

    }

    @Around("dsPoint()")
    public Object around(ProceedingJoinPoint point) throws Throwable{
        DataSource dataSource = getDataSource(point);
        if(StringUtils.isNotNull(dataSource)){
            DynamicDataSourceContextHolder.setDataSourceType(dataSource.value().name());
        }
        try{
            return point.proceed();
        }finally {
            DynamicDataSourceContextHolder.clearDataSourceType();
        }
    }

    private DataSource getDataSource(ProceedingJoinPoint point) {
        MethodSignature signature = (MethodSignature) point.getSignature();
        DataSource dataSource = AnnotationUtils.findAnnotation(signature.getMethod(),DataSource.class);
        if(Objects.nonNull(dataSource)){
            return  dataSource;
        }
        return AnnotationUtils.findAnnotation(signature.getDeclaringType(),DataSource.class);

    }
}
